import datetime
import os
import sys

from selenium import webdriver
from selenium.webdriver.edge.service import Service
from webdriver_manager.microsoft import EdgeChromiumDriverManager
# from selenium.webdriver.edge import webdriver

class Driver:
    driver = ""
    def __init__(self):
        options = webdriver.EdgeOptions()
        self.driver = webdriver.Edge(service=Service(EdgeChromiumDriverManager().install()),options=options)
        self.driver.implicitly_wait(10) #隐式等待两秒
    #添加一个截图函数
    def GetScreeShot(self):
        #在上一级目录创建一个文件夹，以当天的年月日命名
        driname = datetime.datetime.now().strftime("%Y-%m-%d")#把日期格式化一下
        #创建之前先判断一下文件夹存不存在
        if not os.path.exists("../images/"+driname):
            os.mkdir("../images/"+driname)
        #图片的命名要加上调用的函数+年月日+时分秒+png
        filename = sys._getframe().f_back.f_code.co_name+"-"+datetime.datetime.now().strftime("%Y-%m-%d-%H%M%S")+".png"
        self.driver.save_screenshot("../images/"+driname+"/"+filename)
ForumDriver = Driver()#使用一个驱动对象类创建一个驱动对象